class Solution {
public:
    int numberOfBeams(vector<string>& bank) {
        
        int pre=-1;
        int n=bank.size();
        vector<int> vecArr;
        for(int i=0;i<n;++i){
            int cnt=0;
            for(int j=0;j<bank[i].size();++j){
                if(bank[i][j]=='1'){
                    ++cnt;
                }
            }
            if(cnt>0){
                vecArr.push_back(cnt);
            }
        }
        int nRet=0;
        for(int i=1;i<vecArr.size();++i){
            nRet+=vecArr[i]*vecArr[i-1];
        }

        return nRet;
    }
};
